# encoding: utf8
import os, shutil

# ------------ 基本参数 ----------------------
FileType = r'torrent'  # 文件类型
datePrefix = '1021'  # 日期前缀
SOURCE_DIR = ['F:\\迅雷下载', 'D:\\迅雷下载']  # 源文件夹
TARGET_DIR = 'F:\\H_torrent'  # 目标问价夹
KILL_process = 'Thunder.exe'
# ------- _sup File ----支持文件 -------------
f_list = './_sup/fileList.txt'
f_amount = './_sup/_torrentAmount.ini'
f_newList = './_sup/newList.txt'
if not os.path.exists(f_amount): open(f_amount, 'w').write('0')
if os.path.exists(TARGET_DIR): os.system(f'dir')


def get_file_list():
    _fileList = []
    for _dir in SOURCE_DIR: os.system(f'dir /a:-D /b /s /O:D {_dir} >> {f_list}')
    for _t in open(f_list, 'r', encoding='gbk').read().split('\n'):
        if _t.endswith(FileType): _fileList.append(_t)
    os.remove(f_list)
    return _fileList


def _copy(_fl):
    if not os.path.exists(TARGET_DIR): os.mkdir(TARGET_DIR)
    os.popen(f'dir /a:-D /b /s /O:D {TARGET_DIR} >> {f_newList}')
    _am = int(len(open(f_newList, 'r', encoding='gbk').read().split('\n'))) - 1
    for i in range(len(_fl)):
        _am += 1
        _target = f'{TARGET_DIR}\\' + f'{_am}'.rjust(4, '0') + f'.{FileType}'  # Path + FileName + .Filename Extension
        _commend = f'copy "{_fl[i]}" "{_target}"'
        print(_commend)
        # os.system(_commend) # 使用COPY 提示系统找不到指定文件， XCOPY 需要确认 ‘文件/目录’ 且 不管选啥都不过。
        shutil.copy(_fl[i], _target)  # python自带的文件复制方法。
    return


def main():
    fileList = get_file_list()
    for _zz in fileList: print(_zz)
    print(f'共有{len(fileList)}个{FileType}文件')
    # ------- _copy() -----
    _copy(fileList)
    return


if __name__ == '__main__':
    main()
